PYTHONPATH=. torchrun examples/pytorch/text_generation/finetune_text_generation.py \
    --trainer 'text-generation-trainer' \
    --work_dir './tmp' \
    --task 'text2text-generation' \
    --model 'damo/nlp_mt5_zero-shot-augment_chinese-base' \
    --train_dataset_name 'DuReader_robust-QG' \
    --val_dataset_name 'DuReader_robust-QG' \
    --train_split 'train' \
    --val_split 'validation' \
    --src_txt 'text1' \
    --tgt_txt 'text2' \
    --max_epochs 1 \
    --use_model_config True \
    --per_device_train_batch_size 8 \
    --lr 1e-3 \
    --lr_scheduler 'noam' \
